const int n0 = 10, n1 = 12;
template <typename T> using View2D = Kokkos::View<T**, Kokkos::LayoutRight, Kokkos::DefaultExecutionSpace>;
View2D<Kokkos::complex<double> > x("x", n0, n1);
View2D<Kokkos::complex<double> > out_axis0("out_axis0", n0, n1);
View2D<Kokkos::complex<double> > out_axis1("out_axis1", n0, n1);

// Batched 1D FFT along axis 0 (transpose neeed)
KokkosFFT::fft(x, out_axis0, KokkosFFT::FFT_Normalization::BACKWARD, /*axis=*/0);

// Batched 1D FFT along axis 1 (OK without any transpose)
KokkosFFT::fft(x, out_axis0, KokkosFFT::FFT_Normalization::BACKWARD, /*axis=*/1);